# -*- coding: utf-8 -*-
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt

def load_data_set():
    data_mat = []
    label_mat = []
    fr = open('test_set_svm.txt', 'r')
    for line in fr.readlines() :
        line_arr = line.strip().split()
        data_mat.append([float(line_arr[0]), float(line_arr[1])])
        label_mat.append(float(line_arr[2]))
    fr.close()
    return data_mat, label_mat



#data_arr, label_arr = load_data_set()

data_arr = [[1,2],[3,4],[5,6],[7,8],[9,9]]
data_matrix = np.mat(data_arr) 
ret_array = np.ones((np.shape(data_matrix)[0], 1))
ret_array[data_matrix[:,0]>5] = -1000
print(ret_array)

#for i in  range(5):
    #print(data_matrix[i,:])
    #print(data_matrix * data_matrix[i,:].T)
    